from .rul_csv import (PyraformerRUL, ConvTransformerRUL, Seq2SeqRUL, InformerRUL, LSTMRUL, GRURUL, TransformerRUL, CrateRUL,
                  MTCTRUL)

model_dict = {"seq2seq": Seq2SeqRUL,
              "pyraformer": PyraformerRUL,
              "convTransformer": ConvTransformerRUL,
              "informer": InformerRUL,
              "lstm": LSTMRUL,
              "gru": GRURUL,
              "transformer": TransformerRUL,
              "crate": CrateRUL,
              "mtct": MTCTRUL
              }
